昨天提到了利用三軸張量來儲存圖片資料,而今天的筆記主要講的是如何利用張量來儲存多張圖片,一張圖片的儲存主要是藉由 C X H X W 的三軸張量所構成,而在相同概念下我們則利用 N X C X H X W 的四軸張量來儲存多張的圖片資訊,其中N代表著資料集中的圖片數量,而它又稱作此四軸張量的批次軸(batch).

利用張量儲存多張圖片資訊

1. 將所需的模組載入分別為numpy,torch,imageio,os
   
2. 設定一次可載入的圖片數量及利用dtype指定張量資料型別
   
   這裡的張量數值型別我們設定為 torch.uint8,主要是因為相機的攝影格式是以八位元的正整數來表達每一種顏色的強度.
3. 讀取圖檔所在的資料夾路徑,由於我們這裡是要讀取jpg檔,我們必須在程式內以'.jpg'來讀取
   由於我們在第二步設定了圖片的高度與寬度都為256像素,
   而img_t = img_t[:3]這行則是要刪除圖片中可能存在的透明軸.
   

在今天實作的過程中由於我沒有現成的圖片檔,因此我到google上找了幾張圖片來應用,在網路上的圖片像素的值通常都不會是 256 X 256 的格式,因此如果想要順利的應用,那麼就必須要先利用修圖軟體或是網路上能夠更改像素的相關資源對圖片進行修改,我一開始也沒注意到這點導致了錯誤,果然是新手哈哈哈.

參考資料:

• https://www.imagejoy.com/article/655
• https://pytorch.org/docs/stable/tensor_attributes.html